"""
内容：Dense层
日期：2020年6月30日
作者：Howie
"""

from keras.models import Sequential
from keras.layers import Dense
from keras.utils import plot_model

# 超参
INPUT_DIM = 4  # 输入神经元的个数
HIDDEN_LAYER_DIM = [8, 6]  # 输出神经元个数
N_CLASSES = 1  # 二元分类问题
# 解决接收4个输入值并进行二元分类问题的模型
model = Sequential()
model.add(
    Dense(
        units=HIDDEN_LAYER_DIM[0],
        input_dim=INPUT_DIM,
        activation='relu'))
model.add(Dense(units=HIDDEN_LAYER_DIM[1], activation='relu'))
model.add(Dense(units=N_CLASSES, activation='sigmoid'))
# 将搭建的各层以矢量图的形式表现
plot_model(model=model, to_file='./logs/Demo1_model.pdf', show_shapes=True)
